import os
import sys

def create_folders():
    # 定义目标根目录
    target_dir = r"D:\杉达\大数据专业\数据分析与可视化\HeyWhale"

    try:
        # 创建目标根目录（如果不存在）
        if not os.path.exists(target_dir):
            os.makedirs(target_dir)
            print(f"已创建根目录: {target_dir}")
        
        # 中级赛题列表（对应序号2.1-2.5）
        intermediate_competitions = [
            "空气质量预测（PM2.5）",
            "用户贷款违约预测",
            "商品评论情感预测",
            "视频评分预测",
            "风电功率预测"
        ]

        # 高级赛题列表（对应序号3.1-3.5）
        advanced_competitions = [
            "动物图像识别与分类",
            "病理图像组织分类",
            "近地面臭氧浓度预测",
            "热带气旋路径检测",
            "病理图像组织分割"
        ]

        # 创建中级赛题文件夹
        print("\n正在创建中级赛题文件夹...")
        for i, competition in enumerate(intermediate_competitions, 1):
            folder_name = f"{competition} 2.{i}"
            folder_path = os.path.join(target_dir, folder_name)
            if not os.path.exists(folder_path):
                os.makedirs(folder_path)
                print(f"已创建: {folder_name}")
            else:
                print(f"已存在: {folder_name}")

        # 创建高级赛题文件夹
        print("\n正在创建高级赛题文件夹...")
        for i, competition in enumerate(advanced_competitions, 1):
            folder_name = f"{competition} 3.{i}"
            folder_path = os.path.join(target_dir, folder_name)
            if not os.path.exists(folder_path):
                os.makedirs(folder_path)
                print(f"已创建: {folder_name}")
            else:
                print(f"已存在: {folder_name}")

        print("\n所有文件夹创建操作完成！")

    except Exception as e:
        print(f"发生错误: {str(e)}", file=sys.stderr)
        sys.exit(1)

if __name__ == "__main__":
    create_folders()